package 二叉树;

import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

/**
 * @author liucong
 * @date 2021/3/1 - 11:30
 *
 * 给定一个二叉树，返回它的 后序 遍历。
 *
 * 示例:
 *
 * 输入: [1,null,2,3]
 *    1
 *     \
 *      2
 *     /
 *    3
 *
 * 输出: [3,2,1]
 * 进阶: 递归算法很简单，你可以通过迭代算法完成吗？
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/binary-tree-postorder-traversal
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class _145_二叉树的后序遍历 {
    //非递归
    public List<Integer> postorderTraversal2(TreeNode root) {
        List<Integer> list = new ArrayList<>();

        if(root == null) return list;

        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        TreeNode last = null;
        boolean tag = false;
        while(!stack.isEmpty()) {
            TreeNode node = stack.peek();
            if(isLeaf(node) || (tag && isChildren(last, node))){
                last = stack.pop();
                tag = true;
                list.add(last.val);
            }else{
                if(node.right != null){
                    stack.push(node.right);
                }
                if(node.left != null){
                    stack.push(node.left);
                }
            }
        }

        return list;
    }

    private boolean isLeaf(TreeNode node){
        if(node == null) return false;
        if(node.left == null && node.right == null) return true;
        return false;
    }

    private boolean isChildren(TreeNode n1, TreeNode n2){
        if(n1 == n2.left || n1 == n2.right) return true;
        return false;
    }

    //递归
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        postorderTraversal(root, list);
        return list;
    }

    public void postorderTraversal(TreeNode root, List<Integer> list) {
        if(root == null) return;

        postorderTraversal(root.left, list);
        postorderTraversal(root.right, list);
        list.add(root.val);
    }
}
